from .webp_saver import SimpleWebpSaverNode
from .image_resizer import SimpleImageResizerNode
from .text_translator import TextTranslatorNode
from .image_generator import CogViewImageGeneratorNode, GeminiImageGeneratorNode
from .image_contour_generator import ImageContourGeneratorNode
from .batch_reference_loader import BatchReferenceLoaderNode

NODE_CLASS_MAPPINGS = {
    "ImageToWebpSaver": SimpleWebpSaverNode,
    "SimpleImageResizer": SimpleImageResizerNode,
    "TextTranslator": TextTranslatorNode,
    "CogViewImageGenerator": CogViewImageGeneratorNode,
    "GeminiImageGenerator": GeminiImageGeneratorNode,
    "ImageContourGenerator": ImageContourGeneratorNode,
    "BatchReferenceLoader": BatchReferenceLoaderNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "ImageToWebpSaver": f"[Amor] Image WebpSaver",
    "SimpleImageResizer": "[Amor] Image Resizer",
    "TextTranslator": "[Amor] Text Translator",
    "CogViewImageGenerator": "[Amor] CogView Image Generator",
    "GeminiImageGenerator": "[Amor] Gemini Image Generator",
    "ImageContourGenerator": "[Amor] Image Contour Generator",
    "BatchReferenceLoader": "[Amor] Batch Reference Loader",
}
